package com.zhonglai.swagger;

import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;

public class ZlDocket extends Docket{

    public ZlDocket(DocumentationType documentationType) {
        super(documentationType);
    }

    /**
     * 实例化
     * @param enabled 是否启用Swagger
     * @param apiInfo 用来创建该API的基本信息，展示在文档的页面中（自定义展示的信息）
     * @param basePackage 扫描指定包中的swagger注解
     */
    public ZlDocket(boolean enabled, ApiInfo apiInfo,String basePackage) {
         this(DocumentationType.OAS_30);
                // 是否启用Swagger
                this.enable(enabled)
                // 用来创建该API的基本信息，展示在文档的页面中（自定义展示的信息）
                .apiInfo(apiInfo).groupName(apiInfo.getTitle())
                // 设置哪些接口暴露给Swagger展示
                .select()
                // 扫描指定包中的swagger注解
                .apis(RequestHandlerSelectors.basePackage(basePackage))
                // 扫描所有 .apis(RequestHandlerSelectors.any())
                .paths(PathSelectors.any())
                .build();
    }

    public ZlDocket addSecurity()
    {
        SwaggerSecurity swaggerSecurity = new SwaggerSecurity();
        this/* 设置安全模式，swagger可以设置访问token */
                .securitySchemes(swaggerSecurity.securitySchemes())
                .securityContexts(swaggerSecurity.securityContexts());
                return this;
    }

    public ZlDocket pathMapping(String pathMapping)
    {
        if (pathMapping != null && !pathMapping.trim().equals(""))
        {
            this.pathMapping(pathMapping);
        }
        return this;
    }
}
